#!/bin/bash

# Docling Web应用启动脚本

echo "🚀 启动 Docling Web 文档识别应用..."

# 检查虚拟环境
if [ ! -d "venv" ]; then
    echo "❌ 虚拟环境不存在，请先运行 ./install.sh"
    exit 1
fi

# 激活虚拟环境
echo "🔧 激活虚拟环境..."
source venv/bin/activate

# 检查依赖
echo "📚 检查依赖..."
if ! pip show docling > /dev/null 2>&1; then
    echo "❌ Docling 未安装，请先运行 ./install.sh"
    exit 1
fi

# 设置环境变量
if [ -f .env ]; then
    echo "🔧 加载环境变量..."
    export $(cat .env | grep -v '^#' | xargs)
fi

# 设置默认环境变量
export HF_ENDPOINT=${HF_ENDPOINT:-https://hf-mirror.com}
export HF_HOME=${HF_HOME:-./models_cache}
export OMP_NUM_THREADS=${OMP_NUM_THREADS:-4}

# 创建必要目录
mkdir -p uploads
mkdir -p models_cache

# 检查模型
echo "🔍 检查模型状态..."
if [ ! -d "models_cache" ] || [ -z "$(ls -A models_cache)" ]; then
    echo "⚠️ 模型目录为空，建议在Web界面中下载模型"
fi

# 启动应用
echo "🌐 启动Web应用..."
echo "访问地址: http://localhost:5000"
echo "按 Ctrl+C 停止应用"
echo ""

python app.py
